from openai import OpenAI


class DeepSeek:
    def __init__(self, api_key, base_url="https://api.deepseek.com/v1", model="deepseek-chat"):
        self.api_key = api_key
        self.base_url = base_url
        self.model = model
        self.system_prompt = """我希望你充当一个中文翻译、语义纠正和改进者。
        我将用任意语言与你交流，你将识别语言，翻译，并用更正和改进后的中文文本回答我。
        我希望你用更优美、更高雅的词语和句子替换我的句子。
        保持相同的意思，但使它们更具文学性。我希望你只回复更正和改进，别无其他，不要写解释。"""

        self.client = OpenAI(
            api_key=api_key,
            base_url=base_url
        )

    def translate(self, text):
        messages = [
            {"role": "system", "content": self.system_prompt},
            {"role": "user", "content": text}
        ]


        response = self.client.chat.completions.create(
            model=self.model,
            messages=messages,
            temperature=0.1  
        )

        return response.choices[0].message.content
